from selenium.webdriver.common.by import By

from base import BaseAction


class ModelPageHome(BaseAction):
    #     定义元素
    quit_ele = By.CSS_SELECTOR, '.close-btn'
    success_ele = By.XPATH, '//*[text()="0%"]'
    close_ele = By.XPATH, '//span//*[@class="yg-icon-close"]'

    def get_model_quit_ele(self):
        return self.get_element(self.quit_ele)

    def get_model_success_ele(self):
        return self.get_element(self.success_ele)

    def get_model_close_ele(self):
        return self.get_element(self.close_ele)
